package com.xie.leetcode.array;

//118. 杨辉三角
//        给定一个非负整数 numRows，生成「杨辉三角」的前 numRows 行。
//
//        在「杨辉三角」中，每个数是它左上方和右上方的数的和。
//
//
//
//
//
//        示例 1:
//
//        输入: numRows = 5
//        输出: [[1],[1,1],[1,2,1],[1,3,3,1],[1,4,6,4,1]]
//        示例 2:
//
//        输入: numRows = 1
//        输出: [[1]]
//
//
//        提示:
//
//        1 <= numRows <= 30

import java.util.ArrayList;
import java.util.List;

/**
 * @author xiezhendong
 * @date 2021/11/29
 */
public class Generate {

    public static void main(String[] args) {
        Generate generate = new Generate();
        System.out.println(generate.generate(5));
        System.out.println(generate.generate(1));
    }

    public List<List<Integer>> generate(int numRows) {
        List<List<Integer>> values = new ArrayList<>();
        for (int i = 0; i < numRows; i++) {
            if (i <= 1) {
                List<Integer> value = new ArrayList<>();
                for (int i1 = 0; i1 <= i; i1++) {
                    value.add(1);
                }
                values.add(value);
            } else {
                List<Integer> lastValue = values.get(i - 1);

                List<Integer> valueNew = new ArrayList<>();
                for (int j = 0; j <= i; j++) {
                    if (j == 0 || j == i) {
                        valueNew.add(1);
                    } else {
                        valueNew.add(lastValue.get(j - 1) + lastValue.get(j));
                    }
                }
                values.add(valueNew);
            }
        }
        return values;
    }
}
